import os

root_dir = ""

# train_data_path = os.path.join(root_dir, "datas/finished_files/train.bin")
train_data_path = os.path.join(root_dir, "datas/finished_files/chunked/train_*")
eval_data_path = os.path.join(root_dir, "datas/finished_files/val.bin")
decode_data_path = eval_data_path

vocab_path = os.path.join(root_dir, "datas/finished_files/vocab")
log_root = os.path.join(root_dir, "logs")

# Hyperparameters
hidden_dim = 256
emb_dim = 128
batch_size = 32
max_enc_steps = 400
max_dec_steps = 100
beam_size = 4
min_dec_steps = 35
vocab_size = 50000

lr = 0.15
adagrad_init_acc = 0.1
rand_unif_init_mag = 0.02
trunc_norm_init_std = 1e-4
max_grad_norm = 2.0

pointer_gen = True
is_coverage = True
cov_loss_wt = 1.0

eps = 1e-12
max_iterations = 7

use_gpu = True

lr_coverage = 0.15
